using System;
using System.Text;

using Org.BouncyCastle.Utilities;

namespace Org.BouncyCastle.Crypto
{
	/**
	 * super class for all Password Based Encyrption (Pbe) parameter generator classes.
	 */
	public abstract class PbeParametersGenerator
	{
		protected byte[]	mPassword;
		protected byte[]	mSalt;
		protected int		mIterationCount;

		/**
		 * base constructor.
		 */
		protected PbeParametersGenerator()
		{
		}

		/**
		 * initialise the Pbe generator.
		 *
		 * @param password the password converted into bytes (see below).
		 * @param salt the salt to be mixed with the password.
		 * @param iterationCount the number of iterations the "mixing" function
		 * is to be applied for.
		 */
		public virtual void Init(
			byte[]  password,
			byte[]  salt,
			int     iterationCount)
		{
			if (password == null)
				throw new ArgumentNullException("password");
			if (salt == null)
				throw new ArgumentNullException("salt");

			this.mPassword = Arrays.Clone(password);
			this.mSalt = Arrays.Clone(salt);
			this.mIterationCount = iterationCount;
		}

		public virtual byte[] Password
		{
			get { return Arrays.Clone(mPassword); }
		}

		/**
		 * return the password byte array.
		 *
		 * @return the password byte array.
		 */
		[Obsolete("Use 'Password' property")]
		public byte[] GetPassword()
		{
			return Password;
		}

		public virtual byte[] Salt
		{
			get { return Arrays.Clone(mSalt); }
		}

		/**
		 * return the salt byte array.
		 *
		 * @return the salt byte array.
		 */
		[Obsolete("Use 'Salt' property")]
		public byte[] GetSalt()
		{
			return Salt;
		}

		/**
		 * return the iteration count.
		 *
		 * @return the iteration count.
		 */
		public virtual int IterationCount
		{
			get { return mIterationCount; }
		}

		/**
		 * Generate derived parameters for a key of length keySize.
		 *
		 * @param keySize the length, in bits, of the key required.
		 * @return a parameters object representing a key.
		 */
		[Obsolete("Use version with 'algorithm' parameter")]
		public abstract ICipherParameters GenerateDerivedParameters(int keySize);
		public abstract ICipherParameters GenerateDerivedParameters(string algorithm, int keySize);

		/**
		 * Generate derived parameters for a key of length keySize, and
		 * an initialisation vector (IV) of length ivSize.
		 *
		 * @param keySize the length, in bits, of the key required.
		 * @param ivSize the length, in bits, of the iv required.
		 * @return a parameters object representing a key and an IV.
		 */
		[Obsolete("Use version with 'algorithm' parameter")]
		public abstract ICipherParameters GenerateDerivedParameters(int keySize, int ivSize);
		public abstract ICipherParameters GenerateDerivedParameters(string algorithm, int keySize, int ivSize);

		/**
		 * Generate derived parameters for a key of length keySize, specifically
		 * for use with a MAC.
		 *
		 * @param keySize the length, in bits, of the key required.
		 * @return a parameters object representing a key.
		 */
		public abstract ICipherParameters GenerateDerivedMacParameters(int keySize);

		/**
		 * converts a password to a byte array according to the scheme in
		 * Pkcs5 (ascii, no padding)
		 *
		 * @param password a character array reqpresenting the password.
		 * @return a byte array representing the password.
		 */
		public static byte[] Pkcs5PasswordToBytes(
			char[] password)
		{
			return Encoding.ASCII.GetBytes(password);
		}

		public static byte[] Pkcs5PasswordToBytes(
			string password)
		{
			return Encoding.ASCII.GetBytes(password);
		}

		/**
		 * converts a password to a byte array according to the scheme in
		 * PKCS5 (UTF-8, no padding)
		 *
		 * @param password a character array reqpresenting the password.
		 * @return a byte array representing the password.
		 */
		public static byte[] Pkcs5PasswordToUtf8Bytes(
			char[] password)
		{
			return Encoding.UTF8.GetBytes(password);
		}

		public static byte[] Pkcs5PasswordToUtf8Bytes(
			string password)
		{
			return Encoding.UTF8.GetBytes(password);
		}

		/**
		 * converts a password to a byte array according to the scheme in
		 * Pkcs12 (unicode, big endian, 2 zero pad bytes at the end).
		 *
		 * @param password a character array representing the password.
		 * @return a byte array representing the password.
		 */
		public static byte[] Pkcs12PasswordToBytes(
			char[] password)
		{
			return Pkcs12PasswordToBytes(password, false);
		}

		public static byte[] Pkcs12PasswordToBytes(
			char[]	password,
			bool	wrongPkcs12Zero)
		{
			if (password.Length < 1)
			{
				return new byte[wrongPkcs12Zero ? 2 : 0];
			}

			// +1 for extra 2 pad bytes.
			byte[] bytes = new byte[(password.Length + 1) * 2];

			Encoding.BigEndianUnicode.GetBytes(password, 0, password.Length, bytes, 0);

			return bytes;
		}
	}
}
